from llama_index.llms.openai import OpenAI


def moonshot_llm(**kwargs):
    llm = OpenAI(api_key="sk-5D5dtGZGSa1zpqsZFu6DkSYQycwKMbhpahWhkLbQaJUsXkXy",
                 model="moonshot-v1-8k",
                 api_base="https://api.moonshot.cn/v1",
                 **kwargs)
    return llm


def deepseek_llm(**kwargs):
    llm = OpenAI(api_key="sk-0f5de778d709430a964d438c3daf0b6a",
                 model="deepseek-chat",
                 api_base="https://api.deepseek.com/v1",
                 temperature=0.7,
                 **kwargs)
    return llm


def gangtise_llm(**kwargs):
    llm = OpenAI(api_key="sk-Oi1BtzVQEkBIXebvqOAMvicsafbfrOSBFmlfBABpTquLvkxG",
                 model="deepseek-r1-stocklink",
                 api_base="http://114.80.110.181:8138/v1",
                 temperature=0.7,
                 **kwargs)
    return llm


class DeepSeekLLM(OpenAI):
    def __init__(self, **kwargs):
        super().__init__(api_key="sk-5b3ba541a25945a0bc1cf156991447b4",
                         model="deepseek-chat",
                         api_base="https://api.deepseek.com/v1",
                         temperature=0.7,
                         **kwargs)

